package com.czk.tree.binarysorttree.practice;

import lombok.Data;

/**
 * 二叉排序树
 * @Author:ChenZhangKun
 * @Date: 2020/12/5 22:19
 */
public class BinarySortTree {
    public static void main(String[] args) {
        int[] array={7,3,10,12,5,1,9};
        BinarySortTreeDemo binarySortTreeDemo=new BinarySortTreeDemo();
        for (int i : array) {
            binarySortTreeDemo.add(new Node(i));
        }
        binarySortTreeDemo.inorderTraverse();
    }
}
class BinarySortTreeDemo{
    private Node root;

    /**
     * 添加节点
     * @param node
     */
    public void add(Node node){
        if (root==null){
            root=node;
            return;
        }
        root.add(node);
    }

    /**
     * 中序遍历
     */
    public void inorderTraverse(){
        this.root.inOrderTraverse();
    }
}
@Data
class Node{
    public int value;
    public Node left;
    public Node right;

    /**
     * 添加节点
     * @param node
     */
    public void add(Node node){
        // 判断空指针
        if (node==null){
            return;
        }
        if (node.value<this.value){
            // 判断左节点是否为空
            if (this.left==null){
                this.left=node;
            }else {
                // 向左递归
                this.left.add(node);
            }
        }else if (node.value>this.value){
            // 判断右节点是否为空
            if (this.right==null){
                this.right=node;
            }else {
                this.right.add(node);
            }
        }
    }

    /**
     * 查找节点
     * @param value
     * @return
     */
    public Node search(int value){
        // 找到直接返回
        if (value==this.value){
            return this;
        }
        if (value<this.value){
            this.left.search(value);
        }else if (value>this.value){
            this.right.search(value);
        }
        return null;
    }

    /**
     * 查找一个节点的父节点
     * @param value
     * @return
     */
    public Node searchParentNode(int value){
        if ((this.left!=null&&this.left.value==value)||(this.right!=null&&this.right.value==value)){
            return this;
        }
        if (this.left!=null&&this.value>value){
            // 向左递归
            this.left.searchParentNode(value);
        }
        if (this.right!=null&&this.value<value){
            this.right.searchParentNode(value);
        }
        return null;
    }

    public Node(int value) {
        this.value = value;
    }

    /**
     * 中序遍历
     */
    public void inOrderTraverse(){
        if (this.left!=null){
            this.left.inOrderTraverse();
        }
        System.out.println(this.value);
        if (this.right!=null){
            this.right.inOrderTraverse();
        }
    }
}
